from flask import Blueprint,jsonify,request
from db import db1
from jwtc import get_data

bp_courses = Blueprint('courses',__name__)


#部门管理
@bp_courses.route('/deptM',methods=['POST','GET'])
def deptM():
    if request.method == "POST":
        data = get_data()
        #判断名字是否已经存在
        sql = "select id from dept where name='%s'"%(data['name'])
        res = db1.find(sql)
        #存在直接返回
        if res:
            return jsonify({"code":10011,'mes':'已经存在'})
        #不存在添加
        else:
            sql = "insert into dept(name) values('%s')"%(data['name'])
            db1.insert(sql)
            return jsonify({"code":200})
    else:
        sql = "select * from dept"
        res = db1.find(sql)
        return jsonify({"code":200,'list':res})
        

@bp_courses.route('/empM',methods=['POST','GET'])
def empM():
    if request.method == "POST":
        data = get_data()
        #添加员工
        sql = "insert into emp values(0,'%s',%d,%.2f,%d)"%(data['name'],int(data['sex']),float(data['salary']),int(data['dept_id']))
        db1.insert(sql)
        #更新部门表员工总数
        sql1 = 'select total from dept where id=%d'%(int(data['dept_id']))
        res = db1.find(sql1)
        total = int(res['total'])+1
        sql2 = 'update dept set total=%d where id=%d'%(total,int(data['dept_id']))
        db1.insert(sql2)
        return jsonify({"code":200})